
IMAGE_SIZE = [3,512,512] # 图像大小
SUMMARY_WRITER_PATH = './tensorboard' # 训练过程看板路径
LR = 0.01 # 学习率
MAX_EPOCH = 1000 # 最大epoch
BEST_MODEL_PATH = './models/model_best.pth' #最佳模型存储位置